package leetcode;

public class RemoveLinkedListElements203 {
    class ListNode {
        int val;
        ListNode next;

        public ListNode(int val, ListNode next) {
            this.val = val;
            this.next = next;
        }
    }

    /**
     * 设置哨兵节点 通过值删除节点元素
     *
     * @param head 头结点
     * @param val  目标删除的值
     * @return 哨兵节点.next
     */
    public ListNode removeElements(ListNode head, int val) {
        ListNode s = new ListNode(-1, head);
        ListNode p1 = s; // 设一个指针指向哨兵 previous指针
        ListNode p2 = head; // 设一个指针指向哨兵.next 指向当前节点
        while (p2 != null) {
            if (p2.val == val) {
                p1.next = p2.next;
                p2 = p2.next;
            } else {
                p1 = p1.next;
                p2 = p2.next;
            }
        }
        return s.next;
    }

    /**
     * 递归求解 如果是null返回 如果节点值为要删除值返回后面节点作为头指针
     * 如果节点值不相等 则该节点next则为后面与值不相等的节点 并返回该节点
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElements2(ListNode head, int val) {
        if (head == null) {
            return null;
        }
        if (val == head.val) {
            return removeElements2(head.next, val);
        } else {
            head.next = removeElements2(head.next, val);
            return head;
        }
    }
}
